# -*- coding: utf-8 -*-
"""
author:LTH
data:
"""


class GetConfig(object):
    def __init__(self):
        self.project_name = "ArcFace 人脸识别"
        self.use_cuda = True
        self.seed = 123
        self.resume = True
        self.pretrained_weight = "./weight/best.pth"
        self.lr = 0.001
        self.epochs = 200
        self.save_path = "./weight/best.pth"
        self.train_path = "G:/datasets/faces/lfw"
        self.train_batch_size = 64
